package leetcode.dynamic;

/**
 * @author: 26233
 * @create: 2022-03-27 18:22:29
 * @Description
 *
 * 如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为 摆动序列 。第一个差（如果存在的话）可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。
 *
 * 例如， [1, 7, 4, 9, 2, 5] 是一个 摆动序列 ，因为差值 (6, -3, 5, -7, 3) 是正负交替出现的。
 *
 * 相反，[1, 4, 7, 2, 5] 和 [1, 7, 4, 5, 5] 不是摆动序列，第一个序列是因为它的前两个差值都是正数，第二个序列是因为它的最后一个差值为零。
 * 子序列 可以通过从原始序列中删除一些（也可以不删除）元素来获得，剩下的元素保持其原始顺序。
 *
 * 给你一个整数数组 nums ，返回 nums 中作为 摆动序列 的 最长子序列的长度 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：nums = [1,7,4,9,2,5]
 * 输出：6
 * 解释：整个序列均为摆动序列，各元素之间的差值为 (6, -3, 5, -7, 3) 。
 * 示例 2：
 *
 * 输入：nums = [1,17,5,10,13,15,10,5,16,8]
 * 输出：7
 * 解释：这个序列包含几个长度为 7 摆动序列。
 * 其中一个是 [1, 17, 10, 13, 10, 16, 8] ，各元素之间的差值为 (16, -7, 3, -3, 6, -8) 。
 * 示例 3：
 *
 * 输入：nums = [1,2,3,4,5,6,7,8,9]
 * 输出：2
 *  
 *
 * 提示：
 *
 * 1 <= nums.length <= 1000
 * 0 <= nums[i] <= 1000
 *  
 *
 * 进阶：你能否用 O(n) 时间复杂度完成此题?
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/wiggle-subsequence
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * 解法：
 *  1.d[nums.length] 数组的d[i] 表示到i时组成的最长子序列的最后一个值
 *  2.更新d数组：
 *      a. 如果是连续的正差，则期望负差，因此d[i] 应取 max(d[i - 1], nums[i])
 *      b. 如果是连续的负差，则期望正差，因此d[i] 应取 min(d[i - 1], nums[i])
 *      c. 如果交叉，则长度加一，d[i] = nums[i]
 *
 **/

public class LeetCode376_WiggleMaxLength {

    public int wiggleMaxLength(int[] nums) {

        if(nums.length == 1) return 1;
        if(nums.length == 2 && nums[0] != nums[1]) return 2;

        int r;
        int diff;
        int preDiff;

        int[] d  = new int[nums.length];
        d[0] = nums[0];
        r = 1;
        preDiff = 0;
        for (int i = 1; i < nums.length; i++){
            diff = nums[i] - d[i - 1];
            if(diff != 0){
                if((preDiff < 0 && diff > 0) || (preDiff > 0 && diff < 0) || preDiff == 0){
                    r++;
                    d[i] = nums[i];
                    preDiff = diff;
                }else {
                    if(preDiff < 0){
                        d[i] = Math.min(d[i - 1], nums[i]);
                    }else {
                        d[i] = Math.max(d[i - 1], nums[i]);
                    }
                }
            }else {
                d[i] = d[i - 1];
            }
        }

        return r;
    }

    public static void main(String[] args) {
        // int[] nums = new int[]{1,2,3,4,5,6};
        // int[] nums = new int[]{1,17,5,10,13,15,10,5,16,8};
        int[] nums = new int[]{1,3, 20, 17,18};
        System.out.println("r = " + new LeetCode376_WiggleMaxLength().wiggleMaxLength(nums));
    }
}
